package com.dh.leetcode.two;


import org.junit.Assert;
import org.junit.Test;

import java.util.HashMap;

/**
 * @ClassName: _219_contains_duplicate_ii
 * @Description: 219. 存在重复元素 II
 * 给定一个整数数组和一个整数 k，判断数组中是否存在两个不同的索引 i 和 j，使得 nums [i] = nums [j]，并且 i 和 j 的差的 绝对值 至多为 k。
 * <p>
 * 示例 1:
 * <p>
 * 输入: nums = [1,2,3,1], k = 3
 * 输出: true
 * 示例 2:
 * <p>
 * 输入: nums = [1,0,1,1], k = 1
 * 输出: true
 * 示例 3:
 * <p>
 * 输入: nums = [1,2,3,1,2,3], k = 2
 * 输出: false
 * <p>
 * https://leetcode-cn.com/problems/contains-duplicate-ii/
 * @Author: shouzimu
 * @Date: 2021/1/21 16:37
 */
public class _219_contains_duplicate_ii {

    /**
     * 第一版使用O(N*K)的算法超时了
     *
     * 第二次使用O(N)算法
     * 使用map存储值和下标
     *
     * 遇到不为空时则说明第二次遇到，此时判断下标差是否小于K，小于则返回，否则更新下标
     *
     * @param nums
     * @param k
     * @return
     */
    public boolean containsNearbyDuplicate(int[] nums, int k) {
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            Integer index = map.get(nums[i]);
            if (null != index && i - index <= k) {
                return true;
            }
            map.put(nums[i], i);
        }
        return false;
    }

    @Test
    public void containsNearbyDuplicateTest() {
        Assert.assertEquals(true, containsNearbyDuplicate(new int[]{1, 2, 3, 1}, 3));
    }

}


